numpy iteration with nditer